array manipulation hackerrank python